#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2024/4/10 10:32
# @Author  : SunAiJun
# @Site    : 
# @File    : choice_file.py
# @Software: PyCharm
import os


class ChoiceFile:

    @staticmethod
    def choice_file() -> list:
        # 获取当前路径下的所有文件或者文件夹
        files = os.listdir("../")
        # print(files)
        return files

    def get_six_name(self):
        files = self.choice_file()
        for six_name in files:
            if "Six_Stage" in six_name:
                # print(six_name)
                # break
                return six_name

    def get_product_name(self):
        files = self.choice_file()
        for product_name in files:
            if "商品价值" in product_name:
                # print(name)
                # break
                return product_name


if __name__ == '__main__':
    ChoiceFile().get_six_name()
